"""
# 1. 导入unittest包和HTMLTestReport工具
import unittest
from htmltestreport import HTMLTestReport
# 2. 生成测试套件
# 3. 实例化HTMLTestRunner对象
# 4. 执行测试套件


report = HTMLTestReport(file_path, [title], [description])
参数说明：
    file_path： 测试报告文件路径
    title： [可选参数]， 为报告标题， 如XXX测试报告
    description： [可选参数]， 为报告描述信息
# 执行测试套件
report.run(suite)
"""
import unittest
from htmltestreport import HTMLTestReport


# 定义一个加法的函数
def add(a, b):
    return a + b


class TestAdd(unittest.TestCase):
    def test_add01(self):
        print('test_add01111111')
        # 1. 预期，传入参数1,1， 预期结果为2
        # ret为执行结果，实际结果
        ret = add(1, 1)
        print('ret = ', ret)
        # 断言
        self.assertEqual(2, ret)

    def test_add02(self):
        print('test_add02222222')
        # 1. 预期，传入参数10,10， 预期结果为10
        # ret为执行结果，实际结果
        ret = add(10, 10)
        print('ret = ', ret)
        # 断言
        self.assertEqual(10, ret)


if __name__ == '__main__':
    # 如果套件和测试用例在同一个文件，使用命令运行
    # 2. 生成测试套件
    suite = unittest.defaultTestLoader.discover('./', 'hm14*')
    # 3. 实例化HTMLTestReport对象
    report = HTMLTestReport('report.html', '测试报告', '测试报告v1.0版本')
    # 4. 执行测试套件
    report.run(suite)

